[やってみた]Step FunctionsのみでDynamo DBテーブルからアイテム取得/登録をしてみた #reinvent

こんにちは、坂巻です。

re:Invent 2018にて、Step Functionsが新たに8つのマネージドサービスと連携可能になりました！

• [神アップデート]Step Functionsが新たに8つのマネージドサービスと連携可能になりました！ #reinvent

本エントリではStep Functionsから、Dynamo DBより既存アイテムの取得、新アイテムの登録をやってみたいと思います。

やってみた

Dynamo DBテーブル作成

テーブル作成手順については割愛いたしますが、以下のようなテーブルを作成しました。

Step Functions（ステートマシン）作成

タスクステートが2つあるシンプルなステートマシンです。

ステートマシンの定義は以下となります。

{
  "StartAt": "GetItem",
  "States": {
   "GetItem": {
      "Type": "Task",
      "Resource": "arn:aws:states:::dynamodb:getItem",
      "Parameters": {
        "TableName": "TestDB",
        "Key": {
          "ID": {"N": "1"}
        }
      },
      "ResultPath": "$.DynamoDB",
      "Next": "PutItem"
    },
   "PutItem": {
      "Type": "Task",
      "Resource": "arn:aws:states:::dynamodb:putItem",
      "Parameters": {
        "TableName": "TestDB",
        "Item": {
          "ID": {"N": "2"},
          "Name": {"S": "dog"}
        }
      },
      "ResultPath": "$.DynamoDB",
      "End": true
   } 
  }
}

「GetItem」ステートでは、TestDBよりKeyを指定してアイテムを取得し、「PutItem」ステートでは、TestDBにアイテムを登録しています。 データ取得のリソースにはarn:aws:states:::dynamodb:getItemを指定して、データ挿入のリソースにはarn:aws:states:::dynamodb:putItemを指定しています。

Step Functions（ステートマシン）実行

それでは、ステートマシンを実行します。入力イベントはデフォルトのまま実行しました。

ステートマシンの実行が正常終了しました。

Dynamo DBテーブルを確認してみたいと思います。

「PutItem」ステートで指定していたアイテムが登録されていることが確認できました。 次は「GetItem」ステートでアイテムが取得できていたか確認してみます。

「ID」、「Name」が取得できていることが確認できました！！ なお、ステートマシンに付与するIAMロールに、Dynamo DBの権限がないとエラーになりますのでご注意ください。

さいごに

Step Functionsと連携するマネージドサービスが飛躍的に増えたことで、より強力なサービスになったと思います。 今回のように、Step Functions単体でも、DynamoDBテーブルからアイテム取得、登録が可能になりましたので、シンプルな仕組みであれば、Lambdaラッパーなしでもシステムが完成するかもしれませんね！！

参考

• New – Compute, Database, Messaging, Analytics, and Machine Learning Integration for AWS Step Functions